python program to calculate square of a given number